package com.leecode.xiehf.page_02;

import com.leecode.Printer;

/**
 * 编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：
 * <p>
 * 每行中的整数从左到右按升序排列。
 * 每行的第一个整数大于前一行的最后一个整数。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/search-a-2d-matrix
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Solution_0074 extends Printer {

    public static void main(String[] args) {
        Solution_0074 solution = new Solution_0074();
        boolean s = solution.searchMatrix(new int[][]
                {
                        {1, 3, 5, 7},
//                        {1},
//                        {3},
                        {10, 11, 16, 20},
                        {23, 30, 34, 60}
                }, 1);
        System.out.println(s);
    }

    public boolean searchMatrix(int[][] matrix, int target) {
        int height = matrix.length;
        int width = matrix[0].length;

        // 先获取行
        int heightLeft = 0;
        int heightRight = height;
        while (heightLeft < heightRight) {
            int heightMid = (heightLeft + heightRight) / 2;
            if (matrix[heightMid][0] == target || matrix[heightMid][width - 1] == target) {
                return true;
            }
            if (matrix[heightMid][0] < target && matrix[heightMid][width - 1] > target) {
                heightLeft = heightMid;
                break;
            }
            if (matrix[heightMid][0] > target) {
                heightRight = heightMid - 1;
            } else if (matrix[heightMid][width - 1] < target) {
                heightLeft = heightMid + 1;
            }
        }
        if (heightLeft >= height) {
            return false;
        }
        int widthLeft = 0;
        int widthRight = width - 1;
        int[] line = matrix[heightLeft];
        while (widthLeft <= widthRight) {
            int widthMid = (widthLeft + widthRight) / 2;
            if (line[widthMid] == target) {
                return true;
            }
            if (line[widthMid] < target) {
                widthLeft = widthMid + 1;

            } else {
                widthRight = widthMid - 1;
            }
        }
        return false;
    }
}
